package com.atguigu.tingshu.search.service.impl;

import co.elastic.clients.elasticsearch.ElasticsearchClient;
import co.elastic.clients.elasticsearch._types.FieldValue;
import co.elastic.clients.elasticsearch._types.SortOrder;
import co.elastic.clients.elasticsearch._types.aggregations.Aggregate;
import co.elastic.clients.elasticsearch._types.query_dsl.BoolQuery;
import co.elastic.clients.elasticsearch.core.SearchRequest;
import co.elastic.clients.elasticsearch.core.SearchResponse;
import co.elastic.clients.elasticsearch.core.search.Hit;
import co.elastic.clients.elasticsearch.core.search.Suggestion;
import com.alibaba.fastjson.JSON;
import com.atguigu.tingshu.album.client.AlbumInfoFeignClient;
import com.atguigu.tingshu.album.client.CategoryFeignClient;
import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.execption.GuiguException;
import com.atguigu.tingshu.common.login.TsLogin;
import com.atguigu.tingshu.common.result.Result;
import com.atguigu.tingshu.common.util.PinYinUtils;
import com.atguigu.tingshu.model.album.*;
import com.atguigu.tingshu.model.search.AlbumInfoIndex;
import com.atguigu.tingshu.model.search.AttributeValueIndex;
import com.atguigu.tingshu.model.search.SuggestIndex;
import com.atguigu.tingshu.query.search.AlbumIndexQuery;
import com.atguigu.tingshu.search.repository.AlbumIndexRepository;
import com.atguigu.tingshu.search.repository.SuggestIndexRepository;
import com.atguigu.tingshu.search.service.SearchService;
import com.atguigu.tingshu.user.client.UserInfoFeignClient;
import com.atguigu.tingshu.vo.album.AlbumStatVo;
import com.atguigu.tingshu.vo.search.AlbumInfoIndexVo;
import com.atguigu.tingshu.vo.search.AlbumSearchResponseVo;
import com.atguigu.tingshu.vo.user.UserInfoVo;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.statement.create.table.Index;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.data.elasticsearch.core.suggest.Completion;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;

import java.io.IOException;
import java.util.*;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.function.Function;
import java.util.stream.Collectors;


@Slf4j
@Service
@RefreshScope
@SuppressWarnings({"all"})
public class SearchServiceImpl implements SearchService {

    @Autowired
    private AlbumInfoFeignClient albumInfoFeignClient;

    @Autowired
    private ThreadPoolExecutor threadPoolExecutor;

    @Autowired
    private CategoryFeignClient categoryFeignClient;

    @Autowired
    private UserInfoFeignClient userInfoFeignClient;

    @Autowired
    private AlbumIndexRepository albumIndexRepository;

    @Autowired
    private ElasticsearchClient elasticsearchClient;

    @Autowired
    private SuggestIndexRepository suggestIndexRepository;

    @Autowired
    private RedisTemplate  redisTemplate;

    @Value("${ranking.dimension.array}")
    private String rankingDimensionArray;
    @Override
    public void upperAlbum(Long albumId) {
//使用ES索引模型类
        AlbumInfoIndex albumInfoIndex = new AlbumInfoIndex();
        //使用多线程查询三张表的数据给ES索引模型类赋值
        CompletableFuture<AlbumInfo> albumInfoCompletableFuture = CompletableFuture.supplyAsync(() -> {
            //查询专辑信息
            Result<AlbumInfo> albumInfoResult = albumInfoFeignClient.getAlbumInfo(albumId);
            Assert.notNull(albumInfoResult, "查询专辑信息失败");
            AlbumInfo albumInfo = albumInfoResult.getData();
            Assert.notNull(albumInfo, "查询专辑信息失败");
            BeanUtils.copyProperties(albumInfo, albumInfoIndex);
            //获取属性与属性值信息
            List<AlbumAttributeValue> albumAttributeValueVoList = albumInfo.getAlbumAttributeValueVoList();
            if (!CollectionUtils.isEmpty(albumAttributeValueVoList)) {
                //循环遍历
                List<AttributeValueIndex> attributeValueIndexList = albumAttributeValueVoList.stream().map(albumAttributeValue -> {
                    AttributeValueIndex attributeValueIndex = new AttributeValueIndex();
                    attributeValueIndex.setAttributeId(albumAttributeValue.getAttributeId());
                    attributeValueIndex.setValueId(albumAttributeValue.getValueId());
                    return attributeValueIndex;
                }).collect(Collectors.toList());
                albumInfoIndex.setAttributeValueIndexList(attributeValueIndexList);
            }
            return albumInfo;
        }, threadPoolExecutor).exceptionally(throwable -> {
            log.error("查询专辑信息失败：{}", throwable.getMessage());
            return null;
        });


        //  根据三级分类Id 查询 视图就可以了; albumInfo.getCategory3Id(); 查询视图：
        CompletableFuture<Integer> cateCompletableFuture = albumInfoCompletableFuture.thenApplyAsync(albumInfo -> {
            //远程调用根据三级分类Id，获取分类数据;
            Result<BaseCategoryView> categoryViewResult = categoryFeignClient.getBaseCategoryView(albumInfo.getCategory3Id());
            //判断数据
            Assert.notNull(categoryViewResult, "查询视图失败");
            BaseCategoryView baseCategoryView = categoryViewResult.getData();
            Assert.notNull(baseCategoryView, "查询视图失败");
            //赋值
            albumInfoIndex.setCategory1Id(baseCategoryView.getCategory1Id());
            albumInfoIndex.setCategory2Id(baseCategoryView.getCategory2Id());
            return 1;
        }, threadPoolExecutor).exceptionally(throwable -> {
            log.error("查询分类数据失败：{}", throwable.getMessage());
            return 0;
        });

        //查询作者的信息
        CompletableFuture<Integer> userCompletableFuture = albumInfoCompletableFuture.thenApplyAsync(albumInfo -> {
            //  远程调用
            Result<UserInfoVo> userInfoVoResult = userInfoFeignClient.getUserInfoVo(albumInfo.getUserId());
            Assert.notNull(userInfoVoResult, "查询作者信息失败");
            UserInfoVo userInfoVo = userInfoVoResult.getData();
            Assert.notNull(userInfoVo, "查询作者信息失败");
            albumInfoIndex.setAnnouncerName(userInfoVo.getNickname());
            return 1;
        }, threadPoolExecutor).exceptionally(throwable -> {
            log.error("查询分类数据失败：{}", throwable.getMessage());
            return 0;
        });

        //统计信息：播放数、订阅数、购买数、评论数
        CompletableFuture<Integer> statCompletableFuture = albumInfoCompletableFuture.thenApplyAsync(albumInfo -> {
            //远程调用
            Result<AlbumStatVo> statResult = albumInfoFeignClient.getAlbumStatVo(albumId);
            Assert.notNull(statResult, "查询统计信息失败");
            AlbumStatVo statVo = statResult.getData();
            Assert.notNull(statVo, "查询统计信息失败");
            albumInfoIndex.setPlayStatNum(statVo.getPlayStatNum());
            albumInfoIndex.setSubscribeStatNum(statVo.getSubscribeStatNum());
            albumInfoIndex.setBuyStatNum(statVo.getBuyStatNum());
            albumInfoIndex.setCommentStatNum(statVo.getCommentStatNum());
            return 1;
        }, threadPoolExecutor).exceptionally(throwable -> {
            log.error("查询分类数据失败：{}", throwable.getMessage());
            return 0;
        });
        //设置热度排名
        albumInfoIndex.setHotScore(Double.valueOf(new Random().nextInt(10000)));
        try {
            if (null == albumInfoCompletableFuture.get() || 0 == cateCompletableFuture.get() || 0 == userCompletableFuture.get() || 0 == statCompletableFuture.get()) {
                //  获取专辑失败;
                throw new GuiguException(20001, "远程调用获取数据失败");
            }
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        } catch (ExecutionException e) {
            throw new RuntimeException(e);
        }
        //数据汇总：
        albumIndexRepository.save(albumInfoIndex);
        //上架提词库
        SuggestIndex suggestIndex = new SuggestIndex();
        //  赋值：本质赋值的是title;
        suggestIndex.setId(UUID.randomUUID().toString().replace("-", ""));
        suggestIndex.setTitle(albumInfoIndex.getAlbumTitle());
        suggestIndex.setKeyword(new Completion(new String[]{albumInfoIndex.getAlbumTitle()}));
        suggestIndex.setKeywordPinyin(new Completion(new String[]{PinYinUtils.toHanyuPinyin(albumInfoIndex.getAlbumTitle())}));
        suggestIndex.setKeywordSequence(new Completion(new String[]{PinYinUtils.getFirstLetter(albumInfoIndex.getAlbumTitle())}));
        suggestIndexRepository.save(suggestIndex);
        //  介绍;
        SuggestIndex suggestIndexInfo = new SuggestIndex();
        //  赋值：本质赋值的是title;
        suggestIndexInfo.setId(UUID.randomUUID().toString().replace("-", ""));
        suggestIndexInfo.setTitle(albumInfoIndex.getAlbumIntro());
        suggestIndexInfo.setKeyword(new Completion(new String[]{albumInfoIndex.getAlbumIntro()}));
        suggestIndexInfo.setKeywordPinyin(new Completion(new String[]{PinYinUtils.toHanyuPinyin(albumInfoIndex.getAlbumIntro())}));
        suggestIndexInfo.setKeywordSequence(new Completion(new String[]{PinYinUtils.getFirstLetter(albumInfoIndex.getAlbumIntro())}));
        suggestIndexRepository.save(suggestIndexInfo);
        //  作者: 省略...
    }


    public AlbumSearchResponseVo search(AlbumIndexQuery albumIndexQuery) {
        //将业务查询对象转换为Es搜索请求
        SearchRequest searchRequest = this.queryBuildDsl(albumIndexQuery);
        SearchResponse<AlbumInfoIndex> searchResponse = null;
        //使用Es客户端执行搜索请求
        try {
            searchResponse = elasticsearchClient.search(searchRequest, AlbumInfoIndex.class);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        //将 Elasticsearch 的响应转换为业务响应对象
        AlbumSearchResponseVo albumSearchResponseVo = this.parseResultData(searchResponse);
        //设置分页信息
        albumSearchResponseVo.setTotal(searchResponse.hits().total().value());
        albumSearchResponseVo.setPageNo(albumIndexQuery.getPageNo());
        albumSearchResponseVo.setPageSize(albumIndexQuery.getPageSize());
        Long totalPages = (albumSearchResponseVo.getTotal() + albumIndexQuery.getPageSize() - 1) / albumIndexQuery.getPageSize();
        albumSearchResponseVo.setTotalPages(totalPages);
        return albumSearchResponseVo;
    }

    @Override
    public List<String> completeSuggest(String keyword) {
        //  编写: dsl 语句
        SearchResponse<SuggestIndex> searchResponse = null;
        try {
            searchResponse = elasticsearchClient.search(
                    s -> s.index("suggestinfo")
                            .suggest(s1 -> s1.suggesters("suggest_keyword"
                                                    ,
                                                    f -> f.prefix(keyword).completion(c -> c.field("keyword")
                                                            .skipDuplicates(true)
                                                            .fuzzy(f1 -> f1.fuzziness("auto"))
                                                            .size(10))
                                            )
                                            .suggesters("suggest_keywordPinyin"
                                                    ,
                                                    f -> f.prefix(keyword).completion(c -> c.field("keywordPinyin")
                                                            .skipDuplicates(true)
                                                            .fuzzy(f1 -> f1.fuzziness("auto"))
                                                            .size(10))
                                            )
                                            .suggesters("suggest_keywordSequence"
                                                    ,
                                                    f -> f.prefix(keyword).completion(c -> c.field("keywordSequence")
                                                            .skipDuplicates(true)
                                                            .fuzzy(f1 -> f1.fuzziness("auto"))
                                                            .size(10))
                                            )
                            )
                    ,
                    SuggestIndex.class);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        //  获取到查询结果集：
        List<String> list = new ArrayList<>();
        //  定义一个方法：
        list.addAll(this.getSuggestList(searchResponse, "suggest_keyword"));
        list.addAll(this.getSuggestList(searchResponse, "suggest_keywordPinyin"));
        list.addAll(this.getSuggestList(searchResponse, "suggest_keywordSequence"));
        //  返回数据
        return list;
    }

    @Override
    public List<Map<String, Object>> channel(Long category1Id) {
        //调用远程接口通过一级分类id获取三级分类对象List集合
        Result<List<BaseCategory3>> baseCategory3ListResult = categoryFeignClient.findTopBaseCategory3(category1Id);
        //判断
        Assert.notNull(baseCategory3ListResult,"查询三级分类集合失败");
        List<BaseCategory3> baseCategory3List = baseCategory3ListResult.getData();
        Assert.notNull(baseCategory3List, "查询三级分类集合失败");

        //将3级分类对象的list集合转换成Map集合，键：三级分类Id，值：三级分类对象，便于后边的es检索时传入的id获取对应数据
        Map<Long, BaseCategory3> category3Map = baseCategory3List.stream().collect(Collectors.toMap(BaseCategory3::getId, Function.identity()));
        //  List<Long>-->List<FieldValue>将id转换成ES中的FieldValue
        List<FieldValue> fieldValueList = baseCategory3List.stream().map(baseCategory3 -> FieldValue.of(baseCategory3.getId())).collect(Collectors.toList());
        //  专辑集合列表
        SearchResponse<AlbumInfoIndex> searchResponse = null;
        try {
            searchResponse = elasticsearchClient.search(s -> s.index("albuminfo")
                            .query(q -> q.terms(t -> t.field("category3Id")
                                    .terms(t1 -> t1.value(fieldValueList))))
                            .aggregations("agg_category3Id", a -> a.terms(t -> t.field("category3Id"))
                                    .aggregations("topSixHotScoreAgg", a1 -> a1.topHits(t1 -> t1.size(6)
                                            .sort(s1 -> s1.field(f -> f.field("hotScore").order(SortOrder.Desc)))))
                            )
                    ,
                    AlbumInfoIndex.class);
        }catch (IOException e) {
            throw new RuntimeException(e);
        }
        //  获取结果组装数据;
        Aggregate aggregate = searchResponse.aggregations().get("agg_category3Id");
        List<Map<String, Object>> list = aggregate.lterms().buckets().array().stream().map(bucket -> {
                    //  存储数据；需要创建一个map集合：
                    HashMap<String, Object> map = new HashMap<>();
                    //  先获取key;
                    long category3Id = bucket.key();
                    //  存储第一个数据三级分类对象： baseCategory3.name
                    map.put("baseCategory3", category3Map.get(category3Id));
                    //  获取专辑集合列表;
                    Aggregate topSixHotScoreAgg = bucket.aggregations().get("topSixHotScoreAgg");
                    List<AlbumInfoIndex> albumInfoIndexList = topSixHotScoreAgg.topHits().hits().hits().stream().map(hit -> JSON.parseObject(hit.source().toString(), AlbumInfoIndex.class)).collect(Collectors.toList());
                    map.put("list", albumInfoIndexList);
                    //  返回map;
                    return map;
        }).collect(Collectors.toList());
        //返回数据
        return list;
    }

    @Override
    public void updateLatelyAlbumRanking() {
        //  1.  获取所有一级分类数据：select * from base_category1;
        Result<List<BaseCategory1>> listResult = categoryFeignClient.findAllCategory1();
        Assert.notNull(listResult, "查询一级分类数据失败");
        List<BaseCategory1> baseCategory1List = listResult.getData();
        Assert.notNull(baseCategory1List, "查询一级分类数据失败");
        //  获取排序的维度；
        String[] split = rankingDimensionArray.split(",");
        //  循环变量集合 调用es客户端生成dsl语句并执行获取结果集并放入缓存！
        for (BaseCategory1 baseCategory1 : baseCategory1List) {
            //  获取一级分类Id
            Long category1Id = baseCategory1.getId();
            for (String rangking : split) {
                //  调用es api
                SearchResponse<AlbumInfoIndex> searchResponse = null;
                try {
                    searchResponse = elasticsearchClient.search(
                            s -> s.index("albuminfo")
                                    .query(q -> q.term(t -> t.field("category1Id").value(category1Id)))
                                    .sort(s1 -> s1.field(f -> f.field(rangking).order(SortOrder.Desc)))
                            ,
                            AlbumInfoIndex.class);
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
                //  获取集合列表;
                List<AlbumInfoIndex> albumInfoIndexList = searchResponse.hits().hits().stream().map(Hit::source).collect(Collectors.toList());
                //  将每次获取的数据添加到缓存中;
                redisTemplate.opsForHash().put(RedisConstant.RANKING_KEY_PREFIX+category1Id,rangking,albumInfoIndexList);
            }
        }
    }

    @Override
    public List<AlbumInfoIndex> findRankingList(Long category1Id, String rangking) {
       //hget key field
        return (List<AlbumInfoIndex>) redisTemplate.opsForHash().get(RedisConstant.RANKING_KEY_PREFIX+category1Id,rangking);
    }

    /**
     * 获取题词库数据
     *
     * @param searchResponse
     * @param keyword
     * @return
     */
    private Collection<String> getSuggestList(SearchResponse<SuggestIndex> searchResponse, String keyword) {
        //  根据key 获取options:
        List<Suggestion<SuggestIndex>> suggestionList = searchResponse.suggest().get(keyword);
        if (!CollectionUtils.isEmpty(suggestionList)) {
            for (Suggestion<SuggestIndex> suggestIndexSuggestion : suggestionList) {
//            List<String> list = suggestIndexSuggestion.completion().options().stream().map(option -> {
//                String title = option.source().getTitle();
//                return title;
//            }).collect(Collectors.toList());
                return suggestIndexSuggestion.completion().options().stream().map(option -> option.source().getTitle()).collect(Collectors.toList());
            }
        }
        return new ArrayList<>();
    }

    private AlbumSearchResponseVo parseResultData(SearchResponse<AlbumInfoIndex> searchResponse) {
        //创建一个对象
        AlbumSearchResponseVo albumSearchResponseVo = new AlbumSearchResponseVo();
        ArrayList<AlbumInfoIndexVo> list = new ArrayList<>();
        for (Hit<AlbumInfoIndex> hit : searchResponse.hits().hits()){
            //  获取数据：
            AlbumInfoIndex albumInfoIndex = hit.source();
            //  创建一个对象；
            AlbumInfoIndexVo albumInfoIndexVo = new AlbumInfoIndexVo();
            //  属性赋值:
            BeanUtils.copyProperties(albumInfoIndex, albumInfoIndexVo);
            //  判断是否有高亮显示：
            if (null != hit.highlight().get("albumTitle")) {
                //  高亮字段;
                String albumTitle = hit.highlight().get("albumTitle").get(0);
                //  将高亮字段赋值给对象：
                albumInfoIndexVo.setAlbumTitle(albumTitle);
            }
            list.add(albumInfoIndexVo);
        }
        albumSearchResponseVo.setList(list);
        return albumSearchResponseVo;
    }

    private SearchRequest queryBuildDsl(AlbumIndexQuery albumIndexQuery) {
        //SearchRequest 本质就是 DSL 语句封装类;
        SearchRequest.Builder searchRequestBuilder = new SearchRequest.Builder();
        //bool
        BoolQuery.Builder boolQuery = new BoolQuery.Builder();
        //指定索引库;
        searchRequestBuilder.index("albuminfo");
        if (!StringUtils.isEmpty(albumIndexQuery.getKeyword())){
            boolQuery.should(s -> s.match(m -> m.field("albumTitle").query(albumIndexQuery.getKeyword())))
                .should(s -> s.match(m -> m.field("albumIntro").query(albumIndexQuery.getKeyword())))
                .should(s -> s.match(m -> m.field("announcerName").query(albumIndexQuery.getKeyword())));
            //  设置高亮;
            searchRequestBuilder.highlight(h -> h.fields("albumTitle"
                         ,
                            f-> f.preTags("<span color='red'>")
                                    .postTags("</span>")));
        }
        //分类id;
        if (!StringUtils.isEmpty(albumIndexQuery.getCategory1Id())){
            //  根据一级分类Id过滤;
            boolQuery.filter(f -> f.term(t -> t.field("category1Id").value(albumIndexQuery.getCategory1Id())));
        }
        if (!StringUtils.isEmpty(albumIndexQuery.getCategory2Id())) {
            //  根据二级分类Id过滤;
            boolQuery.filter(f -> f.term(t -> t.field("category2Id").value(albumIndexQuery.getCategory2Id())));
        }
        if (!StringUtils.isEmpty(albumIndexQuery.getCategory3Id())) {
            //  根据三级分类Id过滤;
            boolQuery.filter(f -> f.term(t -> t.field("category3Id").value(albumIndexQuery.getCategory3Id())));
        }
        // 用户根据属性和属性值进行过滤
        List<String> attributeList = albumIndexQuery.getAttributeList();
        if (!CollectionUtils.isEmpty(attributeList)) {
            //  循环遍历
            for (String attribute : attributeList) {
                //  attribute = 属性id:属性值id,分割成字符串数组
                String[] split = attribute.split(":");
                if (null != split && split.length == 2) {
                    //  构建dsl 语句:
                    boolQuery.filter(f -> f.nested(n -> n.path("attributeValueIndexList")
                            .query(q -> q.bool(b -> b.must(m -> m.term(t -> t.field("attributeValueIndexList.attributeId").value(split[0])))
                                    .must(m -> m.term(t -> t.field("attributeValueIndexList.valueId").value(split[1])))
                            ))
                    ));
                }
            }
        }

        //分页:
        Integer from = (albumIndexQuery.getPageNo() - 1) * albumIndexQuery.getPageSize();
        searchRequestBuilder.from(from);
        searchRequestBuilder.size(albumIndexQuery.getPageSize());
        //  排序；综合排序[1:desc] 播放量[2:desc] 发布时间[3:desc]；asc:升序 desc:降序
        String order = albumIndexQuery.getOrder();
        //  判断是否有排序功能;
        if (!StringUtils.isEmpty(order)) {
            //  字符串分割;
            String[] split = order.split(":");
            if (null != split && split.length == 2) {
                String field = "";
                switch (split[0]) {
                    case "1":
                        //  综合排序;
                        field = "hotScore";
                        break;
                    case "2":
                        //  播放量排序;
                        field = "playStatNum";
                        break;
                    case "3":
                        //  发布时间排序;
                        field = "createTime";
                }
                //  设置排序字段：
                String finalField = field;
                searchRequestBuilder.sort(s -> s.field(f -> f.field(finalField).order("asc".equals(split[1]) ? SortOrder.Asc : SortOrder.Desc)));
            }
        }
        //  设置bool-->query中
        searchRequestBuilder.query(boolQuery.build()._toQuery());
        SearchRequest searchRequest = searchRequestBuilder.build();
        //  打印DSL语句：
        System.out.println("DSL:\t" + searchRequest.toString());
        return searchRequest;
    }

    @Override
    public void lowerAlbum(Long albumId) {
        albumIndexRepository.deleteById(albumId);
    }

}
